package cc.rengu.oltp.utility.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.math.BigInteger;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.io.IOUtils;

public class MD5Util {
    /***
     * MD5加码 生成32位md5码
     *
     * @throws UnsupportedEncodingException
     */
    public static String string2MD5(String calcString, String charset) throws UnsupportedEncodingException {
        byte[] source = calcString.getBytes(charset);
        String md5String = null;
        char[] hexDigits = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};// 用来将字节转换成16进制表示的字符
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            md.update(source);
            byte[] tmp = md.digest();// MD5 的计算结果是一个 128 位的长整数，
            // 用字节表示就是 16 个字节
            char[] str = new char[16 * 2];// 每个字节用 16 进制表示的话，使用两个字符， 所以表示成 16
            // 进制需要 32 个字符
            int k = 0;// 表示转换结果中对应的字符位置
            for (int i = 0; i < 16; i++) {// 从第一个字节开始，对 MD5 的每一个字节// 转换成 16
                // 进制字符的转换
                byte byte0 = tmp[i];// 取第 i 个字节
                str[k++] = hexDigits[byte0 >>> 4 & 0xf];// 取字节中高 4 位的数字转换,// >>>
                // 为逻辑右移，将符号位一起右移
                str[k++] = hexDigits[byte0 & 0xf];// 取字节中低 4 位的数字转换

            }
            md5String = new String(str);// 换后的结果转换为字符串

        } catch (NoSuchAlgorithmException e) {
            // TODO Auto-generated catch block

        }
        return md5String;
    }

    /**
     * 通过MD5对原始数据进行加密
     *
     * @param origin 原始数据
     * @return 加密后的数据
     * @throws Exception 异常
     */
    public static String MD5Encode(String origin) throws Exception {
        MessageDigest md = MessageDigest.getInstance("MD5");
        String resultString = ByteUtil.bytesToHexString(md.digest(origin.getBytes()));
        return resultString;
    }

    public static String getMd5ByFile(File file) throws FileNotFoundException {
        String value = null;
        FileInputStream in = new FileInputStream(file);
        try {
            MappedByteBuffer byteBuffer = in.getChannel().map(FileChannel.MapMode.READ_ONLY, 0, file.length());
            MessageDigest md5 = MessageDigest.getInstance("MD5");
            md5.update(byteBuffer);
            BigInteger bi = new BigInteger(1, md5.digest());
            value = bi.toString(16);
        } catch (Exception e) {

        } finally {
            if (null != in) {
                try {
                    in.close();
                } catch (IOException e) {

                }
            }
        }
        return value;
    }

    /**
     * 生成MD5加签串
     *
     * @param blockString 加签数据
     * @return 加签字符串
     */
    public static String generateSignString(String... blockString) {
        try {
            StringBuilder stringBuilder = new StringBuilder();
            for (String signBlock : blockString) {
                stringBuilder.append(signBlock);
            }
            String signData = stringBuilder.toString();
            MessageDigest mdInst = MessageDigest.getInstance("MD5");
            mdInst.update(signData.getBytes());
            return ByteUtil.bytesToHexString(mdInst.digest());
        } catch (NoSuchAlgorithmException e) {

            return null;
        }
    }

    /**
     * 校验MD5加签串
     *
     * @param verifyData  校验加签域
     * @param blockString 原始加签数据
     * @return true-成功，false-失败
     */
    public static boolean verifySignString(String verifyData, String... blockString) {
        String signData = generateSignString(blockString);
        if (null == signData) {
            return false;
        } else {
            return verifyData.equalsIgnoreCase(signData);
        }
    }

    public static void main(String[] args) throws IOException {

        String path = "F:\\ftp\\SJQWSD_20161215_1.txt";

        // 第一种实现方式
        String v = getMd5ByFile(new File(path));
        System.out.println("MD5:" + v.toUpperCase());

        // 第二种实现方式
        FileInputStream fis = new FileInputStream(path);
        String md5 = DigestUtils.md5Hex(IOUtils.toByteArray(fis));
        IOUtils.closeQuietly(fis);
        System.out.println("MD5:" + md5);

        // System.out.println("MD5:"+DigestUtils.md5Hex("WANGQIUYUN"));
    }
}